from alg import Solution
from gen import generate_data

def test_min_power_time():
    data = generate_data(3)
    for strength, k in data:
        print("能量需求：", strength, "K：", k)
        result = Solution().minPowerTime(strength, k)
        print(f"最少分钟数: {result}")

def test_case():
    strength = [3, 4, 1]
    k = 1
    print("能量需求：", strength, "K：", k)
    result = Solution().minPowerTime(strength, k)
    print(f"最少分钟数: {result}")

if __name__ == "__main__":
    test_min_power_time()
    test_case()
